import torch

a = 3.258

print(torch.sqrt(torch.tensor(a)))

print(a ** 0.5)
